import Augmentor

p = Augmentor.Pipeline("D:/DIPLOM/augmentation/YOLODataset/images/train")
p.rotate(probability=0.7, max_left_rotation = 25, max_right_rotation = 25)
p.flip_left_right(probability=0.5)
p.random_brightness(probability = 0.5, min_factor=0.4, max_factor=0.9)
p.random_contrast(probability=0.5, min_factor=0.9, max_factor=1.4)
p.random_color(probability=0.5, min_factor=0.4, max_factor=0.9)
p.random_distortion(probability=0.5, grid_width=7, grid_height=8, magnitude=9)
p.sample(200)
